package com.lmb.util;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;

public class ExpiryMap {
    //用来存储数据的容器
    private Map<String,Object>map=new HashMap<String,Object>();
    //存储数据的有效期
    private Map<String,Long> expiryMap=new HashMap<String,Long>();
    //检查过期线程（池）
    private ScheduledExecutorService service= Executors.newSingleThreadScheduledExecutor();//创建一个单线程的线程池
    //检查周期
    private long expiryTime=1000*60;//60秒
    public ExpiryMap(){
        service.scheduleAtFixedRate(this::clearExpiry,expiryTime,expiryTime, TimeUnit.MILLISECONDS);//创建一个定长线程池，支持定时及周期性任务执行
    }//

    public ExpiryMap(long expiryTime){//带有效期的构造方法
        this.expiryTime=expiryTime;//设置有效时间
        service.scheduleAtFixedRate(this::clearExpiry,expiryTime,expiryTime, TimeUnit.MILLISECONDS);//创建一个定长线程池，支持定时及周期性任务执行

    }
    public void put(String key,Object value){//不带有效期的put
        map.put(key,value);//添加Key-Value
        expiryMap.put(key,System.currentTimeMillis()+expiryTime);//添加Key-Value
    }
    public void put(String key,Object value,long expiryTime){//带有效期的put
        map.put(key,value);
        expiryMap.put(key,System.currentTimeMillis()+expiryTime);//添加Key-Value
    }
    public Object get(String key){//获取Key
        if (expiryMap.get(key)!=null && expiryMap.get(key)<System.currentTimeMillis()) {
            map.remove(key);//删除Key-Value
            expiryMap.remove(key);//删除Key-Value
            return null;
        }
        return map.get(key);//返回Key-Value
    }

    public void remove(String key){//删除Key
        map.remove(key);
        expiryMap.remove(key);
    }
    public boolean containsKey(String key){//判断Key是否存在
        if (expiryMap.get(key)<System.currentTimeMillis()){
            map.remove(key);
            expiryMap.remove(key);
        }
        return map.containsKey(key);
    }
    //清除掉过期的Key
    public void clearExpiry(){//线程执行
        for(String key:expiryMap.keySet()){
            if(expiryMap.get(key)<System.currentTimeMillis()){
                map.remove(key);
                expiryMap.remove(key);
            }
        }
    }


}